home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / share / pyshared / DistUpgrade / DistUpgradeQuirks.py < prev    next >
Encoding:
Python Source  |  2009-04-27  |  32.9 KB  |  713 lines
  1. # DistUpgradeQuirks.py 
  2. #  
  3. #  Copyright (c) 2004-2008 Canonical
  4. #  
  5. #  Author: Michael Vogt <michael.vogt@ubuntu.com>
  6. # 
  7. #  This program is free software; you can redistribute it and/or 
  8. #  modify it under the terms of the GNU General Public License as 
  9. #  published by the Free Software Foundation; either version 2 of the
  10. #  License, or (at your option) any later version.
  11. # 
  12. #  This program is distributed in the hope that it will be useful,
  13. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. #  GNU General Public License for more details.
  16. # 
  17. #  You should have received a copy of the GNU General Public License
  18. #  along with this program; if not, write to the Free Software
  19. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20. #  USA
  21.  
  22. import glob
  23. import logging
  24. import os
  25. import os.path
  26. import re
  27. import shutil
  28. import string
  29. import sys
  30. import subprocess
  31. from subprocess import PIPE, Popen, call
  32. from hashlib import md5
  33.  
  34. from DistUpgradeGettext import gettext as _
  35. from DistUpgradeGettext import ngettext
  36. import gettext
  37.  
  38. from computerjanitor.plugin import PluginManager
  39.  
  40. class DistUpgradeQuirks(object):
  41.     """
  42.     This class collects the various quirks handlers that can
  43.     be hooked into to fix/work around issues that the individual
  44.     releases have
  45.     """
  46.     
  47.     def __init__(self, controller, config):
  48.         self.controller = controller
  49.         self._view = controller._view
  50.         self.config = config
  51.         self.uname = Popen(["uname","-r"],stdout=PIPE).communicate()[0].strip()
  52.         self.plugin_manager = PluginManager(self.controller, ["./plugins"])
  53.  
  54.     # the quirk function have the name:
  55.     #  $Name (e.g. PostUpgrade)
  56.     #  $todist$Name (e.g. intrepidPostUpgrade)
  57.     #  $from_$fromdist$Name (e.g. from_dapperPostUpgrade)
  58.     def run(self, quirksName):
  59.         """
  60.         Run the specific quirks handler, the follow handlers are supported:
  61.         - PostInitialUpdate: run *before* the sources.list is rewritten but
  62.                              after a initial apt-get update
  63.         - PostDistUpgradeCache: run *after* the dist-upgrade was calculated
  64.                                 in the cache
  65.         - StartUpgrade: before the first package gets installed (but the
  66.                         download is finished)
  67.         - PostUpgrade: run *after* the upgrade is finished successfully and 
  68.                        packages got installed
  69.         - PostCleanup: run *after* the cleanup (orphaned etc) is finished
  70.         """
  71.         # first check for matching plugins
  72.         for condition in [
  73.             quirksName,
  74.             "%s%s" %  (self.config.get("Sources","To"), quirksName),
  75.             "from_%s%s" % (self.config.get("Sources","From"), quirksName)
  76.             ]:
  77.             for plugin in self.plugin_manager.get_plugins(condition):
  78.                 logging.debug("running quirks plugin %s" % plugin)
  79.                 plugin.do_cleanup_cruft()
  80.         
  81.         # run the handler that is common to all dists
  82.         funcname = "%s" % quirksName
  83.         func = getattr(self, funcname, None)
  84.         if func is not None:
  85.             logging.debug("quirks: running %s" % funcname)
  86.             func()
  87.  
  88.         # run the quirksHandler to-dist
  89.         funcname = "%s%s" % (self.config.get("Sources","To"), quirksName)
  90.         func = getattr(self, funcname, None)
  91.         if func is not None:
  92.             logging.debug("quirks: running %s" % funcname)
  93.             func()
  94.  
  95.         # now run the quirksHandler from_${FROM-DIST}Quirks
  96.         funcname = "from_%s%s" % (self.config.get("Sources","From"), quirksName)
  97.         func = getattr(self, funcname, None)
  98.         if func is not None:
  99.             logging.debug("quirks: running %s" % funcname)
  100.             func()
  101.  
  102.     # individual quirks handler when the dpkg run is finished ---------
  103.     def PostCleanup(self):
  104.         " run after cleanup " 
  105.         logging.debug("running Quirks.PostCleanup")
  106.  
  107.     def from_dapperPostUpgrade(self):
  108.         " this works around quirks for dapper->hardy upgrades "
  109.         logging.debug("running Controller.from_dapperQuirks handler")
  110.         self._rewriteFstab()
  111.         self._checkAdminGroup()
  112.         
  113.     def intrepidPostUpgrade(self):
  114.         " this applies rules for the hardy->intrepid upgrade "
  115.     logging.debug("running Controller.intrepidQuirks handler")
  116.         self._addRelatimeToFstab()
  117.  
  118.     def gutsyPostUpgrade(self):
  119.         """ this function works around quirks in the feisty->gutsy upgrade """
  120.         logging.debug("running Controller.gutsyQuirks handler")
  121.  
  122.     def feistyPostUpgrade(self):
  123.         """ this function works around quirks in the edgy->feisty upgrade """
  124.         logging.debug("running Controller.feistyQuirks handler")
  125.         self._rewriteFstab()
  126.         self._checkAdminGroup()
  127.  
  128.     # quirks when run when the initial apt-get update was run -------
  129.  
  130.     # fglrx is broken in intrepid (no support for xserver 1.5)
  131.     def jauntyPostInitialUpdate(self):
  132.         " quirks that are run before the upgrade to jaunty "
  133.         logging.debug("running %s" %  sys._getframe().f_code.co_name
  134. )
  135.         # this is to deal with the fact that support for some of the cards
  136.         # that fglrx used to support got dropped
  137.         if (self._checkVideoDriver("fglrx") and 
  138.             not self._supportInModaliases("fglrx")):
  139.              res = self._view.askYesNoQuestion(_("Upgrading may reduce desktop "
  140.                                          "effects, and performance in games "
  141.                                          "and other graphically intensive "
  142.                                          "programs."),
  143.                                        _("This computer is currently using "
  144.                                          "the AMD 'fglrx' graphics driver. "
  145.                                          "No version of this driver is "
  146.                                          "available that works with your "
  147.                                          "hardware in Ubuntu "
  148.                                          "9.04.\n\nDo you want to continue?"))
  149.              if res == False:
  150.                  self.controller.abort()
  151.              # if the user wants to continue we remove the fglrx driver
  152.              # here because its no use (no support for this card)
  153.              logging.debug("remove xorg-driver-fglrx,xorg-driver-fglrx-envy,fglrx-kernel-source")
  154.              l=self.controller.config.getlist("Distro","PostUpgradePurge")
  155.              l.append("xorg-driver-fglrx")
  156.              l.append("xorg-driver-fglrx-envy")
  157.              l.append("fglrx-kernel-source")
  158.              l.append("fglrx-amdcccle")
  159.              l.append("xorg-driver-fglrx-dev")
  160.              l.append("libamdxvba1")
  161.              self.controller.config.set("Distro","PostUpgradePurge",",".join(l))
  162.  
  163.     # quirks when the cache upgrade calculation is finished
  164.     def from_dapperPostDistUpgradeCache(self):
  165.         self.hardyPostDistUpgradeCache()
  166.         self.gutsyPostDistUpgradeCache()
  167.         self.feistyPostDistUpgradeCache()
  168.         self.edgyPostDistUpgradeCache()
  169.  
  170.     def jauntyPostDistUpgradeCache(self):
  171.         """ 
  172.         this function works around quirks in the 
  173.         intrepid->jaunty upgrade calculation
  174.         """
  175.         logging.debug("running %s" %  sys._getframe().f_code.co_name)
  176.         # bug 332328 - make sure pidgin-libnotify is upgraded
  177.         for pkg in ["pidgin-libnotify"]:
  178.             if (self.controller.cache.has_key(pkg) and
  179.                 self.controller.cache[pkg].isInstalled and
  180.                 not self.controller.cache[pkg].markedUpgrade):
  181.                 logging.debug("forcing '%s' upgrade" % pkg)
  182.                 self.controller.cache[pkg].markUpgrade()
  183.         # deal with kipi/gwenview/kphotoalbum
  184.         for pkg in ["gwenview","digikam"]:
  185.             if (self.controller.cache.has_key(pkg) and
  186.                 self.controller.cache[pkg].isInstalled and
  187.                 not self.controller.cache[pkg].markedUpgrade):
  188.                 logging.debug("forcing libkipi '%s' upgrade" % pkg)
  189.                 if self.controller.cache.has_key("libkipi0"):
  190.                     logging.debug("removing  libkipi0)")
  191.                     self.controller.cache["libkipi0"].markDelete()
  192.                 self.controller.cache[pkg].markUpgrade()
  193.         
  194.     def intrepidPostDistUpgradeCache(self):
  195.         """ 
  196.         this function works around quirks in the 
  197.         hardy->intrepid upgrade 
  198.         """
  199.         logging.debug("running %s" %  sys._getframe().f_code.co_name)
  200.         # check if a key depends of kubuntu-kde4-desktop is installed
  201.         # and transition in this case as well
  202.         deps_found = False
  203.         if self.config.getlist(frompkg,"KeyDependencies"):
  204.             deps_found = True
  205.             for pkg in self.config.getlist(frompkg,"KeyDependencies"):
  206.                 deps_found &= (self.controller.cache.has_key(pkg) and
  207.                                self.controller.cache[pkg].isInstalled)
  208.         if deps_found:
  209.             logging.debug("transitioning %s to %s (via key depends)" % (frompkg, topkg))
  210.             self.controller.cache[topkg].markInstall()
  211.         # now check for nvidia and show a warning if needed
  212.         cache = self.controller.cache
  213.         for pkgname in ["nvidia-glx-71","nvidia-glx-96"]:
  214.             if (cache.has_key(pkgname) and 
  215.                 cache[pkgname].markedInstall and
  216.                 self._checkVideoDriver("nvidia")):
  217.                 logging.debug("found %s video driver" % pkgname)
  218.                 res = self._view.askYesNoQuestion(_("Upgrading may reduce desktop "
  219.                                         "effects, and performance in games "
  220.                                         "and other graphically intensive "
  221.                                         "programs."),
  222.                                       _("This computer is currently using "
  223.                                         "the NVIDIA 'nvidia' "
  224.                                         "graphics driver. "
  225.                                         "No version of this driver is "
  226.                                         "available that works with your "
  227.                                         "video card in Ubuntu "
  228.                                         "8.10.\n\nDo you want to continue?"))
  229.                 if res == False:
  230.                     self.controller.abort()
  231.                 # if the user continue, do not install the broken driver
  232.                 # so that we can transiton him to the free "nv" one after
  233.                 # the upgrade
  234.                 self.controller.cache[pkgname].markKeep()
  235.         # check if we have sse
  236.         for pkgname in ["nvidia-glx-173","nvidia-glx-177"]:
  237.             if (cache.has_key(pkgname) and 
  238.                 cache[pkgname].markedInstall and
  239.                 self._checkVideoDriver("nvidia")):
  240.                 logging.debug("found %s video driver" % pkgname)
  241.                 if not self._cpuHasSSESupport():
  242.                     logging.warning("nvidia driver that needs SSE but cpu has no SSE support")
  243.                     res = self._view.askYesNoQuestion(_("Upgrading may reduce desktop "
  244.                                         "effects, and performance in games "
  245.                                         "and other graphically intensive "
  246.                                         "programs."),
  247.                                       _("This computer is currently using "
  248.                                         "the NVIDIA 'nvidia' "
  249.                                         "graphics driver. "
  250.                                         "No version of this driver is "
  251.                                         "available that works with your "
  252.                                         "video card in Ubuntu "
  253.                                         "8.10.\n\nDo you want to continue?"))
  254.                     if res == False:
  255.                         self.controller.abort()
  256.                     # if the user continue, do not install the broken driver
  257.                     # so that we can transiton him to the free "nv" one after
  258.                     # the upgrade
  259.                     self.controller.cache[pkgname].markKeep()
  260.         # kdelibs4-dev is unhappy (#279621)
  261.         fromp = "kdelibs4-dev"
  262.         to = "kdelibs5-dev"
  263.         if (self.controller.cache.has_key(fromp) and 
  264.             self.controller.cache[fromp].isInstalled and
  265.             self.controller.cache.has_key(to)):
  266.             self.controller.cache.markInstall(to, "kdelibs4-dev -> kdelibs5-dev transition")
  267.  
  268.     def hardyPostDistUpgradeCache(self):
  269.         """ 
  270.         this function works around quirks in the 
  271.         {dapper,gutsy}->hardy upgrade 
  272.         """
  273.         logging.debug("running %s" %  sys._getframe().f_code.co_name)
  274.         # deal with gnome-translator and help apt with the breaks
  275.         if (self.controller.cache.has_key("nautilus") and
  276.             self.controller.cache["nautilus"].isInstalled and
  277.             not self.controller.cache["nautilus"].markedUpgrade):
  278.             # uninstallable and gutsy apt is unhappy about this
  279.             # breaks because it wants to upgrade it and gives up
  280.             # if it can't
  281.             for broken in ("link-monitor-applet"):
  282.                 if self.controller.cache.has_key(broken) and self.controller.cache[broken].isInstalled:
  283.                     self.controller.cache[broken].markDelete()
  284.             self.controller.cache["nautilus"].markInstall()
  285.         # evms gives problems, remove it if it is not in use
  286.         self._checkAndRemoveEvms()
  287.         # give the language-support-* packages a extra kick
  288.         # (if we have network, otherwise this will not work)
  289.         if self.config.get("Options","withNetwork") == "True":
  290.             for pkg in self.controller.cache:
  291.                 if (pkg.name.startswith("language-support-") and
  292.                     pkg.isInstalled and
  293.                     not pkg.markedUpgrade):
  294.                     self.controller.cache.markInstall(pkg.name,"extra language-support- kick")
  295.  
  296.     def gutsyPostDistUpgradeCache(self):
  297.         """ this function works around quirks in the feisty->gutsy upgrade """
  298.         logging.debug("running %s" %  sys._getframe().f_code.co_name)
  299.         # lowlatency kernel flavour vanished from feisty->gutsy
  300.         try:
  301.             (version, build, flavour) = self.uname.split("-")
  302.             if (flavour == 'lowlatency' or 
  303.                 flavour == '686' or
  304.                 flavour == 'k7'):
  305.                 kernel = "linux-image-generic"
  306.                 if not (self.controller.cache[kernel].isInstalled or self.controller.cache[kernel].markedInstall):
  307.                     logging.debug("Selecting new kernel '%s'" % kernel)
  308.                     self.controller.cache[kernel].markInstall()
  309.         except Exception, e:
  310.             logging.warning("problem while transitioning lowlatency kernel (%s)" % e)
  311.         # fix feisty->gutsy utils-linux -> nfs-common transition (LP: #141559)
  312.         try:
  313.             for line in map(string.strip, open("/proc/mounts")):
  314.                 if line == '' or line.startswith("#"):
  315.                     continue
  316.                 try:
  317.                     (device, mount_point, fstype, options, a, b) = line.split()
  318.                 except Exception, e:
  319.                     logging.error("can't parse line '%s'" % line)
  320.                     continue
  321.                 if "nfs" in fstype:
  322.                     logging.debug("found nfs mount in line '%s', marking nfs-common for install " % line)
  323.                     self.controller.cache["nfs-common"].markInstall()
  324.                     break
  325.         except Exception, e:
  326.             logging.warning("problem while transitioning util-linux -> nfs-common (%s)" % e)
  327.  
  328.     def feistyPostDistUpgradeCache(self):
  329.         """ this function works around quirks in the edgy->feisty upgrade """
  330.         logging.debug("running %s" %  sys._getframe().f_code.co_name)
  331.         # ndiswrapper changed again *sigh*
  332.         for (fr, to) in [("ndiswrapper-utils-1.8","ndiswrapper-utils-1.9")]:
  333.             if self.controller.cache.has_key(fr) and self.controller.cache.has_key(to):
  334.                 if self.controller.cache[fr].isInstalled and not self.controller.cache[to].markedInstall:
  335.                     try:
  336.                         self.controller.cache.markInstall(to,"%s->%s quirk upgrade rule" % (fr, to))
  337.                     except SystemError, e:
  338.                         logging.warning("Failed to apply %s->%s install (%s)" % (fr, to, e))
  339.             
  340.  
  341.     def edgyPostDistUpgradeCache(self):
  342.         """ this function works around quirks in the dapper->edgy upgrade """
  343.         logging.debug("running %s" %  sys._getframe().f_code.co_name)
  344.         for pkg in self.controller.cache:
  345.             # deal with the python2.4-$foo -> python-$foo transition
  346.             if (pkg.name.startswith("python2.4-") and
  347.                 pkg.isInstalled and
  348.                 not pkg.markedUpgrade):
  349.                 basepkg = "python-"+pkg.name[len("python2.4-"):]
  350.                 if (self.controller.cache.has_key(basepkg) and 
  351.                     self.controller.cache[basepkg].candidateDownloadable and
  352.                     not self.controller.cache[basepkg].markedInstall):
  353.                     try:
  354.                         self.controller.cache.markInstall(basepkg,
  355.                                          "python2.4->python upgrade rule")
  356.                     except SystemError, e:
  357.                         logging.debug("Failed to apply python2.4->python install: %s (%s)" % (basepkg, e))
  358.             # xserver-xorg-input-$foo gives us trouble during the upgrade too
  359.             if (pkg.name.startswith("xserver-xorg-input-") and
  360.                 pkg.isInstalled and
  361.                 not pkg.markedUpgrade):
  362.                 try:
  363.                     self.controller.cache.markInstall(pkg.name, "xserver-xorg-input fixup rule")
  364.                 except SystemError, e:
  365.                     logging.debug("Failed to apply fixup: %s (%s)" % (pkg.name, e))
  366.             
  367.         # deal with held-backs that are unneeded
  368.         for pkgname in ["hpijs", "bzr", "tomboy"]:
  369.             if (self.controller.cache.has_key(pkgname) and self.controller.cache[pkgname].isInstalled and
  370.                 self.controller.cache[pkgname].isUpgradable and not self.controller.cache[pkgname].markedUpgrade):
  371.                 try:
  372.                     self.controller.cache.markInstall(pkgname,"%s quirk upgrade rule" % pkgname)
  373.                 except SystemError, e:
  374.                     logging.debug("Failed to apply %s install (%s)" % (pkgname,e))
  375.         # libgl1-mesa-dri from xgl.compiz.info (and friends) breaks the
  376.     # upgrade, work around this here by downgrading the package
  377.         if self.controller.cache.has_key("libgl1-mesa-dri"):
  378.             pkg = self.controller.cache["libgl1-mesa-dri"]
  379.             # the version from the compiz repo has a "6.5.1+cvs20060824" ver
  380.             if (pkg.candidateVersion == pkg.installedVersion and
  381.                 "+cvs2006" in pkg.candidateVersion):
  382.                 for ver in pkg._pkg.VersionList:
  383.                     # the "official" edgy version has "6.5.1~20060817-0ubuntu3"
  384.                     if "~2006" in ver.VerStr:
  385.             # ensure that it is from a trusted repo
  386.             for (VerFileIter, index) in ver.FileList:
  387.                 indexfile = self.controller.cache._list.FindIndex(VerFileIter)
  388.                 if indexfile and indexfile.IsTrusted:
  389.                     logging.info("Forcing downgrade of libgl1-mesa-dri for xgl.compz.info installs")
  390.                                 self.controller.cache._depcache.SetCandidateVer(pkg._pkg, ver)
  391.                     break
  392.                                     
  393.         # deal with general if $foo is installed, install $bar
  394.         for (fr, to) in [("xserver-xorg-driver-all","xserver-xorg-video-all")]:
  395.             if self.controller.cache.has_key(fr) and self.controller.cache.has_key(to):
  396.                 if self.controller.cache[fr].isInstalled and not self.controller.cache[to].markedInstall:
  397.                     try:
  398.                         self.controller.cache.markInstall(to,"%s->%s quirk upgrade rule" % (fr, to))
  399.                     except SystemError, e:
  400.                         logging.debug("Failed to apply %s->%s install (%s)" % (fr, to, e))
  401.                     
  402.     def dapperPostDistUpgradeCache(self):
  403.         """ this function works around quirks in the breezy->dapper upgrade """
  404.         logging.debug("running %s" %  sys._getframe().f_code.co_name)
  405.         if (self.controller.cache.has_key("nvidia-glx") and self.controller.cache["nvidia-glx"].isInstalled and
  406.             self.controller.cache.has_key("nvidia-settings") and self.controller.cache["nvidia-settings"].isInstalled):
  407.             logging.debug("nvidia-settings and nvidia-glx is installed")
  408.             self.controller.cache.markRemove("nvidia-settings")
  409.             self.controller.cache.markInstall("nvidia-glx")
  410.  
  411.     def from_hardyPostDistUpgradeCache(self):
  412.         """ this function works around quirks in upgrades from hardy """
  413.         logging.debug("running %s" %  sys._getframe().f_code.co_name)
  414.         # evms got removed after hardy, warn and abort
  415.         if self._usesEvmsInMounts():
  416.             logging.error("evms in use in /etc/fstab")
  417.             self._view.error(_("evms in use"),
  418.                              _("Your system uses the 'evms' volume manager "
  419.                                "in /proc/mounts. "
  420.                                "The 'evms' software is no longer supported, "
  421.                                "please switch it off and run the upgrade "
  422.                                "again when this is done."))
  423.             self.controller.abort()
  424.  
  425.     # run right before the first packages get installed
  426.     def StartUpgrade(self):
  427.         self._applyPatches()
  428.         self._removeOldApportCrashes()
  429.         self._removeBadMaintainerScripts()
  430.     def jauntyStartUpgrade(self):
  431.         self._createPycentralPkgRemove()
  432.         # hal/NM triggers problem, if the old (intrepid) hal gets
  433.         # triggered for a restart this causes NM to drop all connections
  434.         # because (old) hal thinks it has no devices anymore (LP: #327053)
  435.         ap = "/var/lib/dpkg/info/hal.postinst"
  436.         if os.path.exists(ap):
  437.             # intrepid md5 of hal.postinst (jaunty one is different)
  438.             # md5 jaunty 22c146857d751181cfe299a171fc11c9
  439.             md5sum = "146145275900af343d990a4dea968d7c"
  440.             if md5(open(ap).read()).hexdigest() == md5sum:
  441.                 logging.debug("removing bad script '%s'" % ap)
  442.                 os.unlink(ap)
  443.  
  444.  
  445.     # helpers
  446.     def _removeBadMaintainerScripts(self):
  447.         " remove bad/broken maintainer scripts (last resort) "
  448.         # apache: workaround #95325 (edgy->feisty)
  449.         # pango-libthai #103384 (edgy->feisty)
  450.         bad_scripts = ["/var/lib/dpkg/info/apache2-common.prerm",
  451.                        "/var/lib/dpkg/info/pango-libthai.postrm",
  452.                        ]
  453.         for ap in bad_scripts:
  454.             if os.path.exists(ap):
  455.                 logging.debug("removing bad script '%s'" % ap)
  456.                 os.unlink(ap)
  457.  
  458.     def _createPycentralPkgRemove(self):
  459.         """
  460.         intrepid->jaunty, create /var/lib/pycentral/pkgremove flag file
  461.         to help python-central so that it removes all preinst links
  462.         on upgrade
  463.         """
  464.         logging.debug("adding pkgremove file")
  465.         if not os.path.exists("/var/lib/pycentral/"):
  466.             os.makedirs("/var/lib/pycentral")
  467.         open("/var/lib/pycentral/pkgremove","w")
  468.  
  469.     def _removeOldApportCrashes(self):
  470.         " remove old apport crash files "
  471.         try:
  472.             for f in glob.glob("/var/crash/*.crash"):
  473.                 logging.debug("removing old crash file '%s'" % f)
  474.                 os.unlink(f)
  475.         except Exception, e:
  476.             logging.warning("error during unlink of old crash files (%s)" % e)
  477.  
  478.     def _cpuHasSSESupport(self, cpuinfo="/proc/cpuinfo"):
  479.         " helper that checks if the given cpu has sse support "
  480.         if not os.path.exists(cpuinfo):
  481.             return False
  482.         for line in open(cpuinfo):
  483.             if line.startswith("flags") and not " sse" in line:
  484.                 return False
  485.         return True
  486.  
  487.     def _usesEvmsInMounts(self):
  488.         " check if evms is used in /proc/mounts "
  489.         logging.debug("running _usesEvmsInMounts")
  490.         for line in open("/proc/mounts"):
  491.             line = line.strip()
  492.             if line == '' or line.startswith("#"):
  493.                 continue
  494.             try:
  495.                 (device, mount_point, fstype, options, a, b) = line.split()
  496.             except Exception, e:
  497.                 logging.error("can't parse line '%s'" % line)
  498.                 continue
  499.             if "evms" in device:
  500.                 logging.debug("found evms device in line '%s', skipping " % line)
  501.                 return True
  502.         return False
  503.  
  504.     def _checkAndRemoveEvms(self):
  505.         " check if evms is in use and if not, remove it "
  506.         logging.debug("running _checkAndRemoveEvms")
  507.         if self._usesEvmsInMounts():
  508.             return False
  509.         # if not in use, nuke it
  510.         for pkg in ["evms","libevms-2.5","libevms-dev",
  511.                     "evms-ncurses", "evms-ha",
  512.                     "evms-bootdebug",
  513.                     "evms-gui", "evms-cli",
  514.                     "linux-patch-evms"]:
  515.             if self.controller.cache.has_key(pkg) and self.controller.cache[pkg].isInstalled:
  516.                 self.controller.cache[pkg].markDelete()
  517.         return True
  518.  
  519.     def _addRelatimeToFstab(self):
  520.         " add the relatime option to ext2/ext3 filesystems on upgrade "
  521.         logging.debug("_addRelatime")
  522.         replaced = False
  523.         lines = []
  524.         for line in open("/etc/fstab"):
  525.             line = line.strip()
  526.             if line == '' or line.startswith("#"):
  527.                 lines.append(line)
  528.                 continue
  529.             try:
  530.                 (device, mount_point, fstype, options, a, b) = line.split()
  531.             except Exception, e:
  532.                 logging.error("can't parse line '%s'" % line)
  533.                 lines.append(line)
  534.                 continue
  535.             if (("ext2" in fstype or
  536.                  "ext3" in fstype) and 
  537.                 (not "noatime" in options) and
  538.                 (not "relatime" in options) ):
  539.                 logging.debug("adding 'relatime' to line '%s' " % line)
  540.                 line = line.replace(options,"%s,relatime" % options)
  541.                 logging.debug("replaced line is '%s' " % line)
  542.                 replaced=True
  543.             lines.append(line)
  544.         # we have converted a line
  545.         if replaced:
  546.             logging.debug("writing new /etc/fstab")
  547.             f=open("/etc/fstab.intrepid","w")
  548.             f.write("\n".join(lines))
  549.             # add final newline (see LP: #279093)
  550.             f.write("\n")
  551.             f.close()
  552.             os.rename("/etc/fstab.intrepid","/etc/fstab")
  553.         return True
  554.         
  555.  
  556.     def _rewriteFstab(self):
  557.         " convert /dev/{hd?,scd0} to /dev/cdrom for the feisty upgrade "
  558.         logging.debug("_rewriteFstab()")
  559.         replaced = 0
  560.         lines = []
  561.         # we have one cdrom to convert
  562.         for line in open("/etc/fstab"):
  563.             line = line.strip()
  564.             if line == '' or line.startswith("#"):
  565.                 lines.append(line)
  566.                 continue
  567.             try:
  568.                 (device, mount_point, fstype, options, a, b) = line.split()
  569.             except Exception, e:
  570.                 logging.error("can't parse line '%s'" % line)
  571.                 lines.append(line)
  572.                 continue
  573.             # edgy kernel has /dev/cdrom -> /dev/hd?
  574.             # feisty kernel (for a lot of chipsets) /dev/cdrom -> /dev/scd0
  575.             # this breaks static mounting (LP#86424)
  576.             #
  577.             # we convert here to /dev/cdrom only if current /dev/cdrom
  578.             # points to the device in /etc/fstab already. this ensures
  579.             # that we don't break anything or that we get it wrong
  580.             # for systems with two (or more) cdroms. this is ok, because
  581.             # we convert under the old kernel
  582.             if ("iso9660" in fstype and
  583.                 device != "/dev/cdrom" and
  584.                 os.path.exists("/dev/cdrom") and
  585.                 os.path.realpath("/dev/cdrom") == device
  586.                 ):
  587.                 logging.debug("replacing '%s' " % line)
  588.                 line = line.replace(device,"/dev/cdrom")
  589.                 logging.debug("replaced line is '%s' " % line)
  590.                 replaced += 1
  591.             lines.append(line)
  592.         # we have converted a line (otherwise we would have exited already)
  593.         if replaced > 0:
  594.             logging.debug("writing new /etc/fstab")
  595.             shutil.copy("/etc/fstab","/etc/fstab.edgy")
  596.             f=open("/etc/fstab","w")
  597.             f.write("\n".join(lines))
  598.             # add final newline (see LP: #279093)
  599.             f.write("\n")
  600.             f.close()
  601.         return True
  602.  
  603.     def _checkAdminGroup(self):
  604.         " check if the current sudo user is in the admin group "
  605.         logging.debug("_checkAdminGroup")
  606.         import grp
  607.         try:
  608.             admin_group = grp.getgrnam("admin").gr_mem
  609.         except KeyError, e:
  610.             logging.warning("System has no admin group (%s)" % e)
  611.             subprocess.call(["addgroup","--system","admin"])
  612.         # double paranoia
  613.         try:
  614.             admin_group = grp.getgrnam("admin").gr_mem
  615.         except KeyError, e:
  616.             logging.warning("adding the admin group failed (%s)" % e)
  617.             return
  618.         # if the current SUDO_USER is not in the admin group
  619.         # we add him - this is no security issue because
  620.         # the user is already root so adding him to the admin group
  621.         # does not change anything
  622.         if (os.environ.has_key("SUDO_USER") and
  623.             not os.environ["SUDO_USER"] in admin_group):
  624.             admin_user = os.environ["SUDO_USER"]
  625.             logging.info("SUDO_USER=%s is not in admin group" % admin_user)
  626.             cmd = ["usermod","-a","-G","admin",admin_user]
  627.             res = subprocess.call(cmd)
  628.             logging.debug("cmd: %s returned %i" % (cmd, res))
  629.  
  630.     def _checkVideoDriver(self, name):
  631.         " check if the given driver is in use in xorg.conf "
  632.         XORG="/etc/X11/xorg.conf"
  633.         if not os.path.exists(XORG):
  634.             return False
  635.         for line in open(XORG):
  636.             s=line.split("#")[0].strip()
  637.             # check for fglrx driver entry
  638.             if (s.lower().startswith("driver") and
  639.                 s.endswith('"%s"' % name)):
  640.                 return True
  641.         return False
  642.  
  643.     def _applyPatches(self, patchdir="./patches"):
  644.         """
  645.         helper that applies the patches in patchdir. the format is
  646.         _path_to_file.md5sum
  647.         
  648.         and it will apply the diff to that file if the md5sum
  649.         matches
  650.         """
  651.         if not os.path.exists(patchdir):
  652.             logging.debug("no patchdir")
  653.             return
  654.         if not ("PATH" in os.environ and
  655.                 [p for p in os.environ["PATH"].split(":")
  656.                  if os.path.exists(os.path.join(p,"patch"))]):
  657.             logging.debug("no binary 'patch' found in PATH")
  658.             return
  659.         for f in os.listdir(patchdir):
  660.             # skip, not a patch file, they all end with .$md5sum
  661.             if not "." in f:
  662.                 continue
  663.             logging.debug("check if patch '%s' needs to be applied" % f)
  664.             (encoded_path, md5sum) = string.split(f, ".", 1)
  665.             # FIXME: this is not clever and needs quoting support for
  666.             #        filenames with "_" in the name
  667.             path = encoded_path.replace("_","/")
  668.             if (os.path.exists(path) and
  669.                 md5(open(path).read()).hexdigest() == md5sum):
  670.                 logging.info("applying '%s'" % f)
  671.                 # dry-run first, then patch if ok
  672.                 res = call(["patch","--dry-run","-s","-p0","-i",
  673.                             patchdir+"/"+f])
  674.                 if res == 0:
  675.                     res = call(["patch","-p0","-s","-i",patchdir+"/"+f])
  676.                     logging.info("applied '%s' with %i status" % (f,res))
  677.                 else:
  678.                     logging.warning("dry run failed, ignoring patch '%s'" % f)
  679.                     
  680.     def _supportInModaliases(self, xorgdrivername, modaliasesdir="./modaliases", lspci=None):
  681.         """ 
  682.         Check if xorgdriver will work on this hardware
  683.  
  684.         This helper will check with the modaliasesdir if the given 
  685.         xorgdriver will work on this hardware (or the hardware given
  686.         via the lspci argument)
  687.         """
  688.         if not lspci:
  689.             lspci = set()
  690.             p = subprocess.Popen(["lspci","-n"],stdout=subprocess.PIPE)
  691.             for line in p.communicate()[0].split("\n"):
  692.                 if line:
  693.                     lspci.add(line.split()[2])
  694.         for filename in os.listdir(modaliasesdir):
  695.             for line in open(os.path.join(modaliasesdir,filename)):
  696.                 line = line.strip()
  697.                 if line == "" or line.startswith("#"):
  698.                     continue
  699.                 (key, pciid, xorgdriver, pkgname) = line.split()
  700.                 if xorgdriver != xorgdrivername:
  701.                     continue
  702.                 m = re.match("pci:v0000(.+)d0000(.+)sv.*", pciid)
  703.                 if m:
  704.                     matchid = "%s:%s" % (m.group(1), m.group(2))
  705.                     if matchid.lower() in lspci:
  706.                         logging.debug("found system pciid '%s' in modaliases" % matchid)
  707.                         return True
  708.         logging.debug("checking for %s support in modaliases but none found" % xorgdrivername)
  709.         return False
  710.                     
  711.  
  712.  
  713.